package com.simtop.controller;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.simtop.domain.BaseEntity;
import com.simtop.domain.PageContent;
import com.simtop.utils.ifnull.StringUtils;

import java.util.List;

/**
 * @author lvyongwei
 */
public class BaseController {

    /**
     * 设置请求分页数据
     */
    protected void startPage(BaseEntity baseEntity) {
        Integer pageNum = baseEntity.getPageNum();
        Integer pageSize = baseEntity.getPageSize();
        if (StringUtils.isNotNull(pageNum) && StringUtils.isNotNull(pageSize)) {
            PageHelper.startPage(pageNum, pageSize);
        }
    }

    /**
     * 响应请求分页数据
     */
    protected PageContent getDataTable(List<?> list) {
        Long total = new PageInfo(list).getTotal();
        Integer pageNum = new PageInfo(list).getPageNum();
        Integer pageSize = new PageInfo(list).getPageSize();
        PageContent pageContent = new PageContent();
        pageContent.setList(list);
        pageContent.setTotal(total);
        pageContent.setPageNum(pageNum);
        pageContent.setPageSize(pageSize);
        return pageContent;
    }
}
